shorthand if else python